import os
import pandas as pd

# 定义目录列表
directories = ['pressure', 'rain', 'rainfall_probability', 'temperature', 'wind_speed']
# 假设这是新的文件名列表（需要根据你的实际需求替换）
new_filenames = ['离岛区.csv', '屯门区.csv', '元朗区.csv', '北区.csv', '大埔区.csv', '西贡区.csv', '沙田区.csv']

# 基础路径
base_path = 'data'

# 遍历每个目录
for directory in directories:
    # 完整目录路径
    dir_path = os.path.join(base_path, directory)

    # 获取目录中的所有CSV文件
    csv_files = [f for f in os.listdir(dir_path) if f.endswith('.csv')]

    # 按照某种方式排序文件（例如字母顺序）
    csv_files.sort()

    # 确保CSV文件数量与新文件名数量一致
    if len(csv_files) != len(new_filenames):
        print(f"警告：{directory}目录中的CSV文件数量({len(csv_files)})与新文件名数量({len(new_filenames)})不匹配")
        continue

    # 重命名文件
    for i, old_file in enumerate(csv_files):
        old_path = os.path.join(dir_path, old_file)
        new_path = os.path.join(dir_path, new_filenames[i])
        os.rename(old_path, new_path)
        print(f"已将 {old_path} 重命名为 {new_path}")